Python Lagrange Multipliers